#pragma once

// 协议的制定

#include <iostream>
#include <string>
#include <jsoncpp/json/json.h>

#define MySelf 1

const std::string blank_space_sep = " ";
const std::string protocol_sep = "\n";

// Encode 
std::string Encode(std::string &content)
{
    std::string package = std::to_string(content.size());   // len
    package += protocol_sep;    // len\n
    package += content;         // len\ncontent
    package += protocol_sep;    // len\ncontent\n
    return package;     // "len\ncontent\n"
}
// "len"\n"x op y"\nXXXXXX
// "protocolnumber"\n"len"\n"x op y"\nXXXXXX
bool Decode(std::string &package, std::string *content)
{
    // string -> struct
    st:size_t pos = package.find(protocol_sep);
    if(pos == std::string::npos) return false;
    std::string len_str = package.substr(0, pos);
    std::size_t len = std::stoi(len_str);d:

    // package = len_str + content_str + 2
    std::size_t total_len = len_str.size() + len + 2;
    if(package.size() < total_len) return false;
    
    *content = package.substr(pos+1, len);
    // earse 移除报文 package.erase(0, total_len);
    package.erase(0, total_len);

    return true;
}

// 请求
class Request
{
public:
    int x;
    int y;
    char op;
public:
    Request(int data1, int data2, char oper)
        :x(data1),y(data2),op(oper)
    {
    }
    Request()
    {}
    ~Request()
    {
    }
public:
    // 序列化(struct -> "x op y")
    bool Serialize(std::string *out)
    {
#ifdef MySelf
        // 构建报文的有效载荷
        std::string s = std::to_string(x);
        s += blank_space_sep;
        s += op;
        s += blank_space_sep;
        s += std::to_string(y);
        *out = s;
        return true;
#else
        Json::Value root;
        root["x"] = x;
        root["y"] = y;
        root["op"] = op;
        Json::StyledWriter w;
        *out = w.write(root);
        return true;
#endif
    }

    // 反序列化("x op y -> struct")
    bool Deserialize(const std::string &in)
    {
#ifdef MySelf
        std::size_t left = in.find(blank_space_sep);
        if(left == std::string::npos){
            return false;
        }
        std::string part_x = in.substr(0, left);

        std::size_t right = in.rfind(blank_space_sep);
        if(right == std::string::npos){
            return false;
        }
        std::string part_y = in.substr(right+1);

        // 判断序列是否合法
        if(left + 2 != right){
            return false;
        }

        // 填充 结构体字段
        op = in[left+1];
        x = std::stoi(part_x);
        y = std::stoi(part_y);
        return true;
#else
        Json::Value root;
        Json::Reader r;
        r.parse(in, root);

        x = root["x"].asInt();
        y = root["y"].asInt();
        op = root["op"].asInt();
        return true;
#endif
    }
    void DebugPrint()
    {
        std::cout << "新请求构建完成:  " << x << op << y << "=?" << std::endl;
    }
};


// 响应
class Response
{
public:
    int result;
    int code;
public:
    Response(int ret, int c)
        :result(ret),code(c)
    {}
    Response()
    {}
    ~Response()
    {}
public:
    // 序列化(struct -> "x op y")
    bool Serialize(std::string *out)
    {
#ifdef MySelf   
        // result code
        // 构建响应报文的有效载荷
        std::string s = std::to_string(result);
        s += blank_space_sep;
        s += std::to_string(code);
        s += blank_space_sep;
        *out = s;
        return true;
#else
        Json::Value root;
        root["result"] = result;
        root["code"] = code;

        Json::StyledWriter w;
        *out = w.write(root);
        return true;
#endif
    }
    // 反序列化(""result code" -> struct")
    bool Deserialize(const std::string &in)
    {
#ifdef MySelf
        std::size_t pos = in.find(blank_space_sep);
        if(pos == std::string::npos){
            return false;
        }
        std::string part_res = in.substr(0, pos);
        std::string part_code = in.substr(pos+1);
        
        result = std::stoi(part_res);
        code = std::stoi(part_code);
        return true;
#else
        Json::Value root;
        Json::Reader r;
        r.parse(in, root);

        result = root["result"].asInt();
        code = root["root"].asInt();
        return true;
#endif
    }
    void DebugPrint()
    {
        std::cout << "结果响应完成, result: " << result << ", code: "<< code << std::endl;
    }
};